# -*- coding: utf-8 -*-
# @Author  : longbhu
# @Time    : 2025/3/26 15:19
# @Function:
import pandas as pd

from tif_stats import admin_code_to_name_default

ap_NOx_c = {
    '氮氧化物治理成本':[1111 for _ in range(32)]
}
ap_SO2_c = {
    '二氧化硫治理成本':[1111] * 32
}
ap_PM_c = {
    '烟粉尘治理成本':[1111] * 32
}

wp_COD_c = {
    'COD治理成本':[1111 for _ in range(32)]
}
wp_TP_c = {
    '总磷治理成本':[1111] * 32
}
wp_TN_c = {
    '总氮治理成本':[1111] * 32
}

"""
QapPM 对应 烟粉尘治理成本
QapNOx 对应 氮氧化物治理成本
QapSO2 对应 二氧化硫治理成本
"""


# 创建数据
data = {
    '行政区划代码': list(admin_code_to_name_default.keys()),
    '行政区名称': list(admin_code_to_name_default.values()),
    list(wp_COD_c.keys())[0]: list(wp_COD_c.values())[0],
    list(wp_TN_c.keys())[0]: list(wp_TN_c.values())[0],
    list(wp_TP_c.keys())[0]: list(wp_TP_c.values())[0],
}

# 创建DataFrame
df = pd.DataFrame(data)

# 将DataFrame写入Excel文件
output_path = 'WP价格表.xlsx'
df.to_excel(output_path, index=False)

print(f'Excel文件已保存到 {output_path}')